from typing import Any
from tianshou.utils import BasicLogger


class CustomLogger(BasicLogger):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        print('========= LOG WRITTEN IN %s =========' % self.writer.log_dir)
